package com.by.toolkit.utils;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;
import java.util.concurrent.ThreadLocalRandom;

public class RandomBalanceUtils {

    /**
     * 随机
     *
     * @param items
     * @return
     */
    public static String randomSelect(List<RandomBalanceItem> items) {
        // item 数量
        int length = items.size();
        // 标识所有 item 的权重是否都一样
        boolean sameWeight = true;
        // 用一个数组保存每个 item 的权重
        int[] weights = new int[length];
        // 第一个 item 的权重
        int firstWeight = items.get(0).getWeight();
        weights[0] = firstWeight;
        // 求和总权重
        int totalWeight = firstWeight;
        for (int i = 1; i < length; i++) {
            int weight = items.get(i).getWeight();
            // 保存每个 item 的权重到数组总
            weights[i] = weight;
            // 累加求和总权重
            totalWeight += weight;
            // 如果不是所有 item 的权重都一样，就给标记上 sameWeight = false
            if (sameWeight && weight != firstWeight) {
                sameWeight = false;
            }
        }
        // 计算随机数取到的 item，条件是必须总权重大于0，并且每个 item 的权重都不一样
        if (totalWeight > 0 && !sameWeight) {
            // 基于 0~总数 范围内生成随机数
            int offset = ThreadLocalRandom.current().nextInt(totalWeight);
            // 计算随机数对应的 item
            for (int i = 0; i < length; i++) {
                offset -= weights[i];
                if (offset < 0) {
                    return items.get(i).getItem();
                }
            }
        }
        // 如果所有 item 的权重都一样则随机从 item 列表中返回一个
        return items.get(ThreadLocalRandom.current().nextInt(length)).getItem();
    }

    @Data
    @Builder
    @NoArgsConstructor
    @AllArgsConstructor
    public static class RandomBalanceItem {

        private String item;

        private Integer weight;

    }
}
